package cn.leonis.leetcode;

import java.util.Arrays;

/**
 * @Author Tobi
 * @Date 2021-07-11 上午12:11
 * Merge two Sorted Array 合并两个有序数组
 */
public class MergeSortedArraySolution {

    public static void main(String[] args) {
        int[] nums1 = new int[]{1, 2, 5, 0, 0, 0};
        int[] nums2 = new int[]{2, 3, 6};
        int m = 3, n = 3;
        MergeSortedArraySolution solution = new MergeSortedArraySolution();
        solution.merge2(nums1, m, nums2, n);
        for (int i = 0; i < m + n; i++) {
            System.out.println(nums1[i]);
        }
    }

    public void merge(int[] nums1, int m, int[] nums2, int n) {
        for (int i = 0; i < n; i++) {
            nums1[m + i] = nums2[i];
        }
        Arrays.sort(nums1);
    }

    /**
     * 使用双指针的方式
     * @param nums1
     * @param m
     * @param nums2
     * @param n
     */
    public void merge1(int[] nums1, int m, int[] nums2, int n) {
        int p1 = 0, p2 = 0;
        int[] sorted = new int[m + n];
        int value;
        while (p1 < m || p2 < n) {
            if (p1 == m) {
                value = nums2[p2++];
            }else if (p2 == n) {
                value = nums1[p1++];
            }else if (nums1[p1] < nums2[p2]) {
                value = nums1[p1++];
            }else {
                value = nums2[p2++];
            }

            sorted[p1 + p2 - 1] = value;
        }
        for (int i = 0; i < m + n; i++) {
            nums1[i] = sorted[i];
        }
    }

    /**
     * 使用逆序双指针
     * @param nums1
     * @param m
     * @param nums2
     * @param n
     */
    public void merge2(int[] nums1, int m, int[] nums2, int n) {
        int p1 = m - 1, p2 = n - 1;
        int cursor = m + n - 1;
        int value;
        while (p1 >= 0 || p2 >= 0) {
            if (p1 == -1) {
                value = nums2[p2--];
            }else if (p2 == -1) {
                value = nums1[p1--];
            }else if (nums1[p1] > nums2[p2]) {
                value = nums1[p1--];
            }else {
                value = nums2[p2--];
            }

            nums1[cursor--] = value;
        }
    }
}
